import React, { PureComponent } from 'react'
import { Icon, Badge } from 'antd'
import classNames from 'classnames'
import styles from './index.less'

export default class NoticeIcon extends PureComponent {
  render() {
    const {className, count = 0} = this.props
    const noticeButtonClass = classNames(className, styles.noticeButton)
    return (
      <span className={noticeButtonClass}>
        <Badge count={count}>
          <Icon type="bell" className={styles.icon} />
        </Badge>
      </span>
    )
  }
}
